numpy matplotlib tqdm autograd